history, model  = classification(num_train_samples = 4000, 
                                 learning_rate = 0.001)
plot(history)
model.save('fine_tunning_cats_dogs.h5')
